import torch.nn as nn
import modules

class Model(nn.Module):
    def __init__(self):
        super(Model, self).__init__()
        self.feature =  modules.resnet50(pretrained=True)
        # self.feature.load_state_dict(torch.load(pretrained_url), strict=False )

        self.gazeEs = modules.ResGazeEs()

    def forward(self, x_in):
        features = self.feature(x_in)
        gaze = self.gazeEs(features)
        return gaze


